/*
 * This file Copyright (c) 2015. Walle.
 * (http://www.wallellen.com). All rights reserved.
 *
 *
 * This file is dual-licensed under both the
 * Walle Agreement (WA) and the GNU General Public License.
 * You may elect to use one or the other of these licenses.
 *
 * This file is distributed in the hope that it will be
 * useful, but AS-IS and WITHOUT ANY WARRANTY; without even the
 * implied warranty of MERCHANTABILITY or FITNESS FOR A
 * PARTICULAR PURPOSE, TITLE, or NONINFRINGEMENT.
 * Redistribution, except as permitted by whichever of the GPL
 * or WA you select, is prohibited.
 *
 * 1. For the GPL license (GPL), you can redistribute and/or
 * modify this file under the terms of the GNU General
 * Public License, Version 3, as published by the Free Software
 * Foundation.  You should have received a copy of the GNU
 * General Public License, Version 3 along with this program;
 * if not, write to the Free Software Foundation, Inc., 51
 * Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * 2. For the Walle Agreement (WA), this file
 * and the accompanying materials are made available under the
 * terms of the WA which accompanies this distribution, and
 * is available at http://www.wallellen.com/agreement.html
 *
 *
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER
 */

package com.wallellen.knowledge.web.controller;

import com.wallellen.core.page.ViewMode;
import com.wallellen.core.sql.query.DataQuery;
import com.wallellen.web.WebUtils;
import com.wallellen.web.easy.EasyUiTreeNode;
import com.wallellen.web.easy.EasyUiUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.context.WebApplicationContext;
import org.springframework.web.context.support.WebApplicationContextUtils;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.ServletContext;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpSession;
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;


@RequestMapping("/frame")
@Controller
public class FrameController extends WebUtils {
    private final static Logger log = LoggerFactory.getLogger(FrameController.class);

    @RequestMapping("/index")
    public ModelAndView index(HttpServletRequest request, HttpSession session) {
        return ViewMode.getInstance().putAttr("menus", getCurrentUserMenus(session)).returnModelAndView("frame/frame");
    }

    @SuppressWarnings("unchecked")
    @RequestMapping("/service")
    @ResponseBody
    public List<EasyUiTreeNode> allUrl(DataQuery query, HttpServletRequest request) {
        log.info("正式系统请 关闭该服务");
        ServletContext servletContext = request.getSession().getServletContext();
        WebApplicationContext appContext = WebApplicationContextUtils.getWebApplicationContext(servletContext);
        List<Map<String, Object>> list = new ArrayList<Map<String, Object>>();
        // 获取所有的RequestMapping
        Map<String, Object> allRequestMappings = appContext.getBeansWithAnnotation(Controller.class);
        int n = 1;
        for (Object obj : allRequestMappings.values()) {
            RequestMapping classRequest = obj.getClass().getAnnotation(RequestMapping.class);
            if (classRequest != null) {
                // 封装父节点
                Map<String, Object> superNode = new HashMap<String, Object>();
                int m = ++n;
                superNode.put("SID", n);
                superNode.put("PID", "none");
                if (classRequest.value() == null || classRequest.value().length == 0) {
                    superNode.put("NA", "NoGroup");
                } else {
                    superNode.put("NA", classRequest.value()[0]);
                }
                list.add(superNode);
                for (Method method : obj.getClass().getMethods()) {
                    RequestMapping methodRequest = method.getAnnotation(RequestMapping.class);
                    if (methodRequest != null) {
                        // 封装子节点
                        Map<String, Object> childeNode = new HashMap<String, Object>();
                        n++;
                        childeNode.put("SID", n);
                        childeNode.put("PID", m);

                        if (classRequest.value() == null || classRequest.value().length == 0) {
                            childeNode.put("NA", methodRequest.value()[0].replaceAll("/", ""));
                        } else {
                            childeNode.put("NA",
                                    classRequest.value()[0].replaceAll("/", "") + methodRequest.value()[0]);
                        }

                        list.add(childeNode);
                    }
                }
            }
        }
        return (List<EasyUiTreeNode>) ViewMode.returnListObjMode(EasyUiUtils.formatAjaxTree(list, "PID", "SID", "NA"));
    }
}
